Square root of a matrix